package com.sh.data.engine.domain.util;

import com.sh.data.engine.common.enumDefinition.DSType;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;

/**
 * @author: mengzheng.mhc
 * @date: 2024/6/13 11:56
 */
@Slf4j
public class DBUtil {

    public static boolean checkRDBMS(String dsType) {
        if (StringUtils.isBlank(dsType)) {
            return false;
        }
        if (dsType.equals(DSType.MySQL.name())
            || dsType.equals(DSType.SQLServer.name())
            || dsType.equals(DSType.PostgreSQL.name())
            || dsType.equals(DSType.MatrixDB.name())
            || dsType.equals(DSType.DB2.name())
            || dsType.equals(DSType.Oracle.name())
            || dsType.equals(DSType.Sap.name())
            || dsType.equals(DSType.DM.name())
            || dsType.equals(DSType.KINGBASE8.name())
            || dsType.equals(DSType.TiDB.name())) {
            return true;
        }
        return false;
    }

    public static Integer getOraclePort(String url) {
        if (url == null || url.length() == 0) {
            return null;
        }
        try {
            String strAt = "@";
            String str2Point = ":";
            String str = "/";
            if (url.contains(strAt)) {
                String hostPort = url.substring(url.indexOf(strAt));
                if (hostPort.contains(str2Point)) {
                    String port = hostPort.substring(hostPort.indexOf(str2Point) + 1);
                    if (port.contains(str2Point)) {
                        return Integer.parseInt(port.substring(0, port.indexOf(str2Point)));
                    } else if (port.contains(str)) {
                        return Integer.parseInt(port.substring(0, port.indexOf(str)));
                    }
                }
            }
        } catch (Exception e) {
            log.error("obtain oracle port error ", e);
        }
        return null;
    }
}
